//
//  Problem1379.swift
//  TestProject
//
//  Created by 武侠 on 2021/6/8.
//  Copyright © 2021 zhulong. All rights reserved.
//

import UIKit

/*
 1379. 找出克隆二叉树中的相同节点
 给你两棵二叉树，原始树 original 和克隆树 cloned，以及一个位于原始树 original 中的目标节点 target。

 其中，克隆树 cloned 是原始树 original 的一个 副本 。

 请找出在树 cloned 中，与 target 相同 的节点，并返回对该节点的引用（在 C/C++ 等有指针的语言中返回 节点指针，其他语言返回节点本身）。

  

 注意：
     你 不能 对两棵二叉树，以及 target 节点进行更改。
     只能 返回对克隆树 cloned 中已有的节点的引用。
     进阶：如果树中允许出现值相同的节点，你将如何解答？
 示例 1:
     输入: tree = [7,4,3,null,null,6,19], target = 3
     输出: 3
     解释: 上图画出了树 original 和 cloned。target 节点在树 original 中，用绿色标记。答案是树 cloned 中的黄颜色的节点（其他示例类似）。
 示例 2:
     输入: tree = [7], target =  7
     输出: 7
 示例 3:
     输入: tree = [8,null,6,null,5,null,4,null,3,null,2,null,1], target = 4
     输出: 4
 示例 4:
     输入: tree = [1,2,3,4,5,6,7,8,9,10], target = 5
     输出: 5
 示例 5:
     输入: tree = [1,2,null,3], target = 2
     输出: 2
 提示：
     树中节点的数量范围为 [1, 10^4] 。
     同一棵树中，没有值相同的节点。
     target 节点是树 original 中的一个节点，并且不会是 null 。
 */
@objcMembers class Problem1379: NSObject {
    func solution() {
        let root1 = TreeNode(7)
        root1.left = TreeNode(4)
        root1.right = TreeNode(3)
        root1.right?.left = TreeNode(6)
        root1.right?.right = TreeNode(19)
        
        let root2 = TreeNode(7)
        root2.left = TreeNode(4)
        root2.right = TreeNode(3)
        root2.right?.left = TreeNode(6)
        root2.right?.right = TreeNode(19)
        
    }
    
    func getTargetCopy(_ original: TreeNode?, _ cloned: TreeNode?, _ target: TreeNode?) -> TreeNode? {
        if original == nil {
            return nil
        }
        if original === target {
            return cloned
        }
        
        let leftNode = getTargetCopy(original?.left, cloned?.left, target)
        if leftNode === target {
            return leftNode
        }
        let rightNode = getTargetCopy(original?.right, cloned?.right, target)
        if rightNode === target {
            return rightNode
        }
        return nil
    }
}
